import torch

print(f"torch版本号:  --{torch.__version__}")  # 查看torch当前版本号

print(f"cuda版本号:   --{torch.version.cuda}")  # 编译当前版本的torch使用的cuda版本号

print(f"cuda是否可用:  --{torch.cuda.is_available()}")  # 查看当前cuda是否可用于当前版本的Torch，如果输出True，则表示可用

print(torch.cuda.current_device())

print(torch.cuda.get_device_name(0))